

# 检查 Pytorch
import torch, torchvision
print('Pytorch 版本', torch.__version__)
# print('cuda 版本', torch.cuda._version)
print('CUDA 是否可用',torch.cuda.is_available())


# 检查 mmcv
import mmcv
from mmcv.ops import get_compiling_cuda_version, get_compiler_version
print('MMCV版本', mmcv.__version__)
print('CUDA版本', get_compiling_cuda_version())
print('编译器版本', get_compiler_version())